class DefaultConfig():
    # backbone
    pretrained = True
    # freeze_stage_1 = True
    freeze_bn = True                #小bachsize，冻结BN层

    # fpn
    fpn_out_channels = 128
    use_p5 = True

    # head
    class_num = 16
    use_GN_head = True
    prior = 0.01
    add_centerness = True
    cnt_on_reg = True

    # training
    strides = [8, 16, 32, 64, 128]
    limit_range = [[-1, 64], [64, 128], [128, 256], [256, 512], [512, 999999]]
    # inference
    score_threshold = 0.01         #越小map越高
    # score_threshold = 0.1
    nms_iou_threshold = 0.5
    max_detection_boxes_num = 1000

    # efficiennet
    efficientNet = 'efficientnet-b0'
    EFFICIENTDET = {
        'efficientnet-b0': {'input_size': 512,
                            'backbone': 'B0',
                            'W_bifpn': 96,
                            'D_bifpn': 1, },
        'efficientnet-b1': {'input_size': 640,
                            'backbone': 'B1',
                            'W_bifpn': 128,
                            'D_bifpn': 1, },
        'efficientnet-b2': {'input_size': 512,
                            'backbone': 'B2',
                            'W_bifpn': 128,
                            'D_bifpn': 1},  # 'D_bifpn': 4
        'efficientnet-b3': {'input_size': 640,
                            'backbone': 'B3',
                            'W_bifpn': 160,
                            'D_bifpn': 1},  # 'D_bifpn': 5
        'efficientnet-b4': {'input_size': 768,
                            'backbone': 'B4',
                            'W_bifpn': 224,
                            'D_bifpn': 1},
        'efficientnet-b5': {'input_size': 1280,
                            'backbone': 'B5',
                            'W_bifpn': 196,
                            'D_bifpn': 7},
        'efficientnet-b6': {'input_size': 1408,
                            'backbone': 'B6',
                            'W_bifpn': 384,
                            'D_bifpn': 8},
        'efficientnet-b7': {'input_size': 1636,
                            'backbone': 'B6',
                            'W_bifpn': 384,
                            'D_bifpn': 8},
    }


